tensor([[0.6200, 0.6200, 0.6200, 0.6200, 0.6200],
        [0.3800, 0.3800, 0.3800, 0.3800, 0.3800]], grad_fn=<RoundBackward1>)